from utils import csv_to_fasta, retrive_esm1b_embedding, mutate_single_seq_ECs, compute_esm_distance


csv_to_fasta("/state/partition/wzzheng/clean/data/datasets/new.csv", "/state/partition/wzzheng/clean/data/datasets/new.fasta")
retrive_esm1b_embedding(fasta_name="/state/partition/wzzheng/clean/data/datasets/new")

# train_file = "split100_train_split_0"

# train_fasta_file = mutate_single_seq_ECs(train_file, data_dir=data_dir)
# retrive_esm1b_embedding(train_fasta_file, data_dir=data_dir)
# compute_esm_distance(train_file)

# train_file = "split100_train_split_1"

# train_fasta_file = mutate_single_seq_ECs(train_file, data_dir=data_dir)
# retrive_esm1b_embedding(train_fasta_file, data_dir=data_dir)
# compute_esm_distance(train_file)

# train_file = "split100_train_split_2"

# train_fasta_file = mutate_single_seq_ECs(train_file, data_dir=data_dir)
# retrive_esm1b_embedding(train_fasta_file, data_dir=data_dir)
# compute_esm_distance(train_file)

# train_file = "split100_train_split_3"

# train_fasta_file = mutate_single_seq_ECs(train_file, data_dir=data_dir)
# retrive_esm1b_embedding(train_fasta_file, data_dir=data_dir)
# compute_esm_distance(train_file)

# train_file = "split100_train_split_4"

# train_fasta_file = mutate_single_seq_ECs(train_file, data_dir=data_dir)
# retrive_esm1b_embedding(train_fasta_file, data_dir=data_dir)
# compute_esm_distance(train_file)